package com.atguigu.tingshu.search.config;

import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * @author zh
 * @date 2025/9/4
 * @description
 */
@Component
public class ThreadPoolExecutorConfig {
    //  自定义线程池;
    @Bean
    public ThreadPoolExecutor threadPoolExecutor() {
        //  自定义线程池; 因为使用工具类创建的线程池
        //  要么是阻塞队列
        //  要么是最大线程个数默认是：INTEGER.MAX_AVALUE(21亿) 造成OOM!
        //  获取当前cpu 核数
        int cpuCoreNum = Runtime.getRuntime().availableProcessors();
        System.out.println("cpuCoreNum:" + cpuCoreNum);
        return new ThreadPoolExecutor(
                //核心线程数据; 业务需求： io=2n cpu=n+1
                cpuCoreNum,
                //最大线程数据
                cpuCoreNum * 2,
                //存活时间
                3,
                //单位
                TimeUnit.SECONDS,
                //阻塞队列
                new ArrayBlockingQueue<>(100),
                //线程工厂
                Executors.defaultThreadFactory(),
                //拒绝策略
                new ThreadPoolExecutor.AbortPolicy()
        );
    }
}
